home *** CD-ROM | disk | FTP | other *** search
/ Chip: 2001 Haziran / CHIP Haziran2001.iso / prog / haziran / 19 / setup.exe / data.z / drvload.cpp < prev    next >
C/C++ Source or Header  |  2001-04-11  |  16KB  |  601 lines
  1. #include <windows.h>
  2. #include <stdio.h>
  3. #include "drvload.h"
  4.  
  5. LoadDriver::LoadDriver()
  6. {
  7.     m_h = NULL;
  8.     ClearErr();
  9. }
  10.  
  11. LoadDriver::~LoadDriver()
  12. {
  13. }
  14.  
  15. LoadDriver95::LoadDriver95()
  16. {
  17.     if (!OpenSCMHandle())
  18.         PrintErr("Cannot open service control manager\n");
  19. }
  20.  
  21. LoadDriver95::~LoadDriver95()
  22. {
  23.     if (m_h)
  24.         CloseSCMHandle();
  25. }
  26.  
  27. LoadDriver98::LoadDriver98()
  28. {
  29.     if (!OpenSCMHandle())
  30.         PrintErr("Cannot open service control manager\n");
  31. }
  32.  
  33. LoadDriver98::~LoadDriver98()
  34. {
  35.     //if (m_h)
  36.     //    CloseSCMHandle();
  37. }
  38.  
  39. LoadDriverNT::LoadDriverNT()
  40. {
  41.     if (!OpenSCMHandle())
  42.         PrintErr("Cannot open service control manager\n");
  43. }
  44.  
  45. LoadDriverNT::~LoadDriverNT()
  46. {
  47.     if (m_h)
  48.         CloseSCMHandle();
  49. }
  50.  
  51. BOOL LoadDriver::Init(char *sDriverName, char *sDriverFile, DWORD startLevel, char *sDriverPath, BOOL fUseReg)
  52. {
  53.     if (!m_h)
  54.         return FALSE;
  55.     m_fUseReg = fUseReg;
  56.     m_iStartLevel = startLevel;
  57.     SetPaths (sDriverName, sDriverFile, sDriverPath);
  58.     return TRUE;
  59. }
  60.  
  61. char *LoadDriver::GetLastMsg()
  62. { 
  63.     return m_sLastMsg;
  64. }
  65.  
  66. void LoadDriver::ClearErr()
  67. {
  68.     m_sLastMsg[0] = '\0';
  69.     m_fHasErr = FALSE;
  70.     m_fHasWrn = FALSE;
  71. }
  72.  
  73. void LoadDriver::PrintErr(char *msg)
  74. {
  75.     strcat (m_sLastMsg, msg);
  76.     m_fHasErr = TRUE;
  77. }
  78.  
  79. void LoadDriver::PrintWrn(char *msg)
  80. {
  81.     strcat (m_sLastMsg, msg);
  82.     m_fHasWrn = TRUE;
  83. }
  84.  
  85. BOOL LoadDriver::HasErr()
  86. {
  87.     return m_fHasErr;
  88. }
  89.  
  90. BOOL LoadDriver::HasWrn()
  91. {
  92.     return m_fHasWrn;
  93. }
  94.  
  95. BOOL LoadDriver::HasMsg()
  96. {
  97.     return m_fHasWrn || m_fHasErr;
  98. }
  99.  
  100. // VXDLDR error codes
  101. #define VXDLDR_ERR_OUT_OF_MEMORY    1
  102. #define VXDLDR_ERR_IN_DOS       2
  103. #define VXDLDR_ERR_FILE_OPEN_ERROR  3
  104. #define VXDLDR_ERR_FILE_READ        4
  105. #define VXDLDR_ERR_DUPLICATE_DEVICE 5
  106. #define VXDLDR_ERR_BAD_DEVICE_FILE  6
  107. #define VXDLDR_ERR_DEVICE_REFUSED   7
  108. #define VXDLDR_ERR_NO_SUCH_DEVICE   8
  109. #define VXDLDR_ERR_DEVICE_UNLOADABLE    9
  110. #define VXDLDR_ERR_ALLOC_V86_AREA   10
  111. #define VXDLDR_ERR_BAD_API_FUNCTION 11
  112.  
  113. #define VXDLDR_ERR_MAX          11
  114.  
  115. // VXDLDR driver functions
  116. #define VXDLDR_APIFUNC_GETVERSION   0
  117. #define VXDLDR_APIFUNC_LOADDEVICE   1
  118. #define VXDLDR_APIFUNC_UNLOADDEVICE 2
  119.  
  120. BOOL LoadDriver95::OpenSCMHandle()
  121. {
  122.     if (m_h)
  123.         return FALSE;
  124.     m_h = CreateFile(
  125.         "\\\\.\\VXDLDR",        // device driver name
  126.         0,              // access
  127.         0,              // share mode
  128.         0,              // security attributes
  129.         CREATE_NEW,             // create options
  130.         FILE_FLAG_DELETE_ON_CLOSE,  // flags
  131.         0);             // template
  132.  
  133.     if (m_h==INVALID_HANDLE_VALUE)
  134.         m_h = NULL;
  135.     return m_h!=NULL;
  136. }
  137.  
  138. void LoadDriver95::CloseSCMHandle()
  139. {
  140.     CloseHandle(m_h);
  141.     m_h = NULL;
  142. }
  143.  
  144. BOOL LoadDriver98::OpenSCMHandle()
  145. {
  146.     // currently starting and stoping not supported OpenSCMHandle is empty
  147.     m_h = (HANDLE) 1; // a 'dummy' handle
  148.     return TRUE;
  149. }
  150.  
  151. void LoadDriver98::CloseSCMHandle()
  152. {
  153. }
  154.  
  155. BOOL LoadDriverNT::OpenSCMHandle()
  156. {
  157.     m_h = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  158.     return m_h!=NULL;
  159. }
  160.  
  161. void LoadDriverNT::CloseSCMHandle()
  162. {
  163.     CloseServiceHandle(m_h);
  164.     m_h = NULL;
  165. }
  166.  
  167. void LoadDriver::SetPaths(char *sDriverName, char *sDriverFile, char *sDriverPath)
  168. {
  169.     strcpy (m_sDriverName, sDriverName);
  170.     strcpy (m_sDriverFile, sDriverFile);
  171.     if (sDriverPath)
  172.         strcpy (m_sDriverPath, sDriverPath);
  173.     else strcpy (m_sDriverPath,"");
  174. }
  175.  
  176. void LoadDriver95::SetPaths(char *sDriverName, char *sDriverFile, char *sDriverPath)
  177. {
  178.     LoadDriver::SetPaths(sDriverName, sDriverFile, sDriverPath);
  179.     if( !sDriverPath)
  180.     {
  181.         char sSystemPath[1024];
  182.         GetSystemDirectory( sSystemPath,sizeof( sSystemPath) );
  183.         if( sSystemPath[ strlen(sSystemPath)-1] != '\\' ) 
  184.             strcat(sSystemPath,"\\");
  185.         sprintf (m_sDriverPath,"%sVmm32\\%s.VXD", sSystemPath, m_sDriverFile);
  186.     }
  187. }
  188.  
  189. void LoadDriver98::SetPaths(char *sDriverName, char *sDriverFile, char *sDriverPath)
  190. {
  191.     LoadDriver::SetPaths(sDriverName, sDriverFile, sDriverPath);
  192.     if( !sDriverPath)
  193.         sprintf (m_sDriverPath, "\\SystemRoot\\System32\\drivers\\%s.SYS", sDriverFile);
  194. }
  195.  
  196. void LoadDriverNT::SetPaths(char *sDriverName, char *sDriverFile, char *sDriverPath)
  197. {
  198.     LoadDriver::SetPaths(sDriverName, sDriverFile, sDriverPath);
  199.     if( !sDriverPath)
  200.         sprintf (m_sDriverPath, "\\SystemRoot\\System32\\drivers\\%s.SYS", sDriverFile);
  201. }
  202.  
  203. BOOL LoadDriver95::Create()
  204. {
  205.     ClearErr();
  206.  
  207.     char sKey[1024];
  208.     sprintf (sKey, "System\\CurrentControlSet\\Services\\VxD\\%s", m_sDriverName);
  209.     // updating registry
  210.     HKEY hKey;
  211.     ULONG rc;
  212.     RegCreateKeyEx (HKEY_LOCAL_MACHINE, sKey,0,"",0,KEY_ALL_ACCESS, NULL, &hKey, &rc);
  213.     if (rc==REG_OPENED_EXISTING_KEY)
  214.     {
  215.         PrintWrn (m_sDriverName);
  216.         PrintWrn (" already exists in Win-95 registry\n");
  217.     }
  218.     else if (rc!=REG_CREATED_NEW_KEY && rc!=ERROR_SUCCESS)
  219.     {
  220.         PrintErr("Cannot open registry key\n");
  221.         return FALSE;
  222.     }
  223.  
  224.     RegSetValueEx (hKey, "Start", 0, REG_BINARY, (PBYTE) "\0", 1);
  225.     char sVal[1024];
  226.     sprintf (sVal, "*%s", m_sDriverFile);
  227.     RegSetValueEx (hKey, "StaticVxD", 0, REG_SZ, (PBYTE) sVal, 9);
  228.  
  229.     RegCloseKey (hKey);
  230.  
  231.     return TRUE;
  232. }
  233.  
  234. BOOL LoadDriver95::Delete()
  235. {
  236.     ClearErr();
  237.     LONG rc;
  238.     char sKey[1024];
  239.     sprintf (sKey, "System\\CurrentControlSet\\Services\\VxD\\%s", m_sDriverName);
  240.     rc = RegDeleteKey (HKEY_LOCAL_MACHINE, sKey);
  241.     if (rc!=ERROR_SUCCESS)
  242.     {
  243.         PrintErr (m_sDriverName);
  244.         PrintErr (" does not exist in Win-95 registry\n");
  245.         return FALSE;
  246.     }
  247.     return TRUE;
  248. }
  249.  
  250. BOOL LoadDriver98::Create()
  251. {
  252.     ClearErr();
  253.  
  254.     char sKey[1024];
  255.     sprintf (sKey, "System\\CurrentControlSet\\Services\\%s", m_sDriverName);
  256.     // updating registry
  257.     HKEY hKey;
  258.     ULONG rc;
  259.     RegCreateKeyEx (HKEY_LOCAL_MACHINE, sKey,0,"",0,KEY_ALL_ACCESS, NULL, &hKey, &rc);
  260.     if (rc==REG_OPENED_EXISTING_KEY)
  261.     {
  262.         PrintWrn (m_sDriverName);
  263.         PrintWrn (" already exists in Win-98 registry\n");
  264.     }
  265.     else if (rc!=REG_CREATED_NEW_KEY && rc!=ERROR_SUCCESS)
  266.     {
  267.         PrintErr("Cannot open registry key\n");
  268.         return FALSE;
  269.     }
  270.  
  271.  
  272.     DWORD val = 2;
  273.     RegSetValueEx (hKey, "Start", 0, REG_BINARY, (PBYTE) &m_iStartLevel, 4);
  274.     val = 1;
  275.     RegSetValueEx (hKey, "Type", 0, REG_BINARY, (PBYTE) &val, 4);
  276.     val = 1;
  277.     RegSetValueEx (hKey, "ErrorControl", 0, REG_BINARY, (PBYTE) &val, 4);
  278. //    RegSetValueEx (hKey, "Group", 0, REG_SZ, (PBYTE) "Extended Base", 14);
  279.     RegSetValueEx (hKey, "ImagePath", 0, REG_SZ, (PBYTE) m_sDriverPath, 41);
  280.     RegSetValueEx (hKey, "DisplayName", 0, REG_SZ, (PBYTE) m_sDriverName, 10);
  281.  
  282.     RegCloseKey (hKey);
  283.  
  284.     return TRUE;
  285. }
  286.  
  287. BOOL LoadDriver98::Delete()
  288. {
  289.     ClearErr();
  290.     LONG rc;
  291.     char sKey[1024];
  292.     sprintf (sKey, "System\\CurrentControlSet\\Services\\%s", m_sDriverName);
  293.     rc = RegDeleteKey (HKEY_LOCAL_MACHINE, sKey);
  294.     if (rc!=ERROR_SUCCESS)
  295.     {
  296.         PrintErr (m_sDriverName);
  297.         PrintErr (" does not exist in Win-98 registry\n");
  298.         return FALSE;
  299.     }
  300.     return TRUE;
  301. }
  302.  
  303. BOOL LoadDriverNT::CreateReg()
  304. {
  305.     ClearErr();
  306.  
  307.     char sKey[1024];
  308.     sprintf (sKey, "System\\CurrentControlSet\\Services\\%s", m_sDriverName);
  309.     // updating registry
  310.     HKEY hKey;
  311.     ULONG rc;
  312.     RegCreateKeyEx (HKEY_LOCAL_MACHINE, sKey,0,"",0,KEY_ALL_ACCESS, NULL, &hKey, &rc);
  313.     if (rc==REG_OPENED_EXISTING_KEY)
  314.     {
  315.         PrintWrn (m_sDriverName);
  316.         PrintWrn (" already exists in Win-NT registry\n");
  317.     }
  318.     else if (rc!=REG_CREATED_NEW_KEY && rc!=ERROR_SUCCESS)
  319.     {
  320.         PrintErr("Cannot open registry key\n");
  321.         return FALSE;
  322.     }
  323.  
  324.     DWORD val = 2;
  325.     RegSetValueEx (hKey, "Start", 0, REG_DWORD, (PBYTE) &m_iStartLevel, 4);
  326.     val = 1;
  327.     RegSetValueEx (hKey, "Type", 0, REG_DWORD, (PBYTE) &val, 4);
  328.     val = 1;
  329.     RegSetValueEx (hKey, "ErrorControl", 0, REG_DWORD, (PBYTE) &val, 4);
  330.     RegSetValueEx (hKey, "Group", 0, REG_SZ, (PBYTE) "Extended Base", 14);
  331.     RegSetValueEx (hKey, "ImagePath", 0, REG_SZ, (PBYTE) m_sDriverPath, 41);
  332.     RegSetValueEx (hKey, "DisplayName", 0, REG_SZ, (PBYTE) m_sDriverName, 10);
  333.  
  334.     RegCloseKey (hKey);
  335.  
  336.     return TRUE;
  337. }
  338.  
  339. BOOL LoadDriverNT::DeleteReg()
  340. {
  341.     ClearErr();
  342.     LONG rc;
  343.     char sKey[1024];
  344.     sprintf (sKey, "System\\CurrentControlSet\\Services\\%s", m_sDriverName);
  345.     rc = RegDeleteKey (HKEY_LOCAL_MACHINE, sKey);
  346.     if (rc!=ERROR_SUCCESS)
  347.     {
  348.         PrintErr (m_sDriverName);
  349.         PrintErr (" does not exist in Win-NT registry\n");
  350.         return FALSE;
  351.     }
  352.     return TRUE;
  353. }
  354.  
  355. BOOL LoadDriver95::Start()
  356. {
  357.     ULONG status, nRet;
  358.  
  359.     if (!m_h)
  360.         return FALSE;
  361.  
  362.     status = DeviceIoControl(
  363.         m_h,                // handle
  364.         VXDLDR_APIFUNC_LOADDEVICE,  // control code (see vxdldr.h in DDK)
  365.         (void *) m_sDriverPath,     // path of vxd file (No \\.\ prefix)
  366.         strlen(m_sDriverPath)+1,    // input size
  367.         NULL,               // output buffer
  368.         0,              // output buffer size
  369.         &nRet,              // receives count returned
  370.         NULL);
  371.     return status==1; 
  372. }
  373.  
  374. BOOL LoadDriver95::Stop()
  375. {
  376.     typedef struct {
  377.             LONG    vxdid;       // device ID of VxD to unload, or -1 to use name
  378.             CHAR    vxdname[256];// null terminated, case sensitive name of VxD 
  379.                                  //   to unload if VxD ID is -1
  380.     } UNLOAD_INFO;
  381.     int i;
  382.             
  383.     UNLOAD_INFO unloadInfo;
  384.     ULONG status, nRet;
  385.  
  386.     if (!m_h)
  387.         return FALSE;
  388.  
  389.     unloadInfo.vxdid = -1;
  390.     strcpy (unloadInfo.vxdname, m_sDriverFile);
  391.     for (i=0; unloadInfo.vxdname[i]; i++)
  392.         unloadInfo.vxdname[i] = toupper(unloadInfo.vxdname[i]);
  393.  
  394.     for (i=0; i<1000; i++)
  395.     {
  396.         status = DeviceIoControl(
  397.             m_h,                // handle
  398.             VXDLDR_APIFUNC_UNLOADDEVICE,    // control code (see vxdldr.h in DDK)
  399.             (void *) &unloadInfo,       // path of vxd file (No \\.\ prefix)
  400.             sizeof (unloadInfo),        // input size
  401.             NULL,               // output buffer
  402.             0,              // output buffer size
  403.             &nRet,              // receives count returned
  404.             NULL);
  405.         if (status!=1)
  406.             break;
  407.     }
  408.     if (i==0)
  409.     {
  410.         PrintWrn("Could not unload driver, or driver was previously loaded.\n");
  411.     }
  412.     return i>0 && i<1000; 
  413. }
  414.  
  415. BOOL LoadDriver98::Start()
  416. {
  417.     PrintWrn("Dynamic Loading of (SYS) driver not available for WIN 98.\n"
  418.          "This is a Windows 98 limitation. Please reboot the system.\n"
  419.          "For a reboot free installation on 98, use the -VXD option\n");
  420.     return FALSE;
  421. }
  422.  
  423. BOOL LoadDriver98::Stop()
  424. {
  425.     PrintWrn("Dynamic unloading of (SYS)driver not available for WIN 98.\n"
  426.          "This is a Windows 98 limitation. Please reboot the system.\n"
  427.          "For a reboot free installation on 98, use the -VXD option\n");
  428.     return FALSE;
  429. }
  430.  
  431. BOOL LoadDriverNT::Create()
  432. {
  433.     SC_HANDLE  sr;
  434.     BOOL fOk = TRUE;
  435.  
  436.     if (m_fUseReg)
  437.         return CreateReg();
  438.  
  439.     if (!m_h)
  440.         return FALSE;
  441.  
  442.     sr = CreateService(
  443.         m_h,    // handle to service control manager database  
  444.         m_sDriverName,  // pointer to name of service to start 
  445.         m_sDriverName,  // pointer to display name 
  446.         SERVICE_ALL_ACCESS, // type of access to service 
  447.         SERVICE_KERNEL_DRIVER,  // type of service 
  448.         m_iStartLevel,  // when to start service 
  449.         SERVICE_ERROR_NORMAL,   // severity if service fails to start 
  450.         m_sDriverPath,  // pointer to name of binary file 
  451.         NULL,   // pointer to name of load ordering group 
  452.         NULL,   // pointer to variable to get tag identifier 
  453.         NULL,   // pointer to array of dependency names 
  454.         NULL,   // pointer to account name of service 
  455.         NULL    // pointer to password for service account 
  456.     );  
  457.     if (sr==NULL) fOk = FALSE;
  458.  
  459.     if (sr) CloseServiceHandle (sr);
  460.  
  461.     return fOk;
  462. }
  463.  
  464. BOOL LoadDriverNT::Start()
  465. {
  466.     SC_HANDLE  sr;
  467.     BOOL fOk = TRUE;
  468.  
  469.     if (!m_h)
  470.         return FALSE;
  471.  
  472.     sr = OpenService(m_h ,m_sDriverName,SERVICE_ALL_ACCESS );
  473.  
  474.     if (sr==NULL)
  475.     {
  476.         fOk = FALSE;
  477.     }
  478.     else
  479.     {
  480.         BOOL rc = StartService (
  481.             sr,    // service identifier
  482.             0,             // number of arguments
  483.             NULL           // pointer to arguments
  484.         );
  485.  
  486.         if (!rc) fOk = FALSE;
  487.     }
  488.  
  489.     if (sr) CloseServiceHandle (sr);
  490.  
  491.     return fOk;
  492. }
  493.  
  494. BOOL LoadDriverNT::Stop()
  495. {
  496.     SC_HANDLE  sr;
  497.     BOOL fOk = TRUE;
  498.     SERVICE_STATUS  serviceStatus;
  499.  
  500.     sr = OpenService(m_h ,m_sDriverName ,SERVICE_ALL_ACCESS );
  501.  
  502.     if (sr==NULL) 
  503.         fOk = FALSE;
  504.  
  505.     if (sr)
  506.     {
  507.         BOOL rc = ControlService ( sr ,SERVICE_CONTROL_STOP ,&serviceStatus );
  508.         if( !rc )
  509.         {
  510.             if( ERROR_SERVICE_NOT_ACTIVE == GetLastError() )
  511.                 rc = TRUE;
  512.         }
  513.  
  514.         if( SERVICE_STOPPED != serviceStatus.dwCurrentState)
  515.             rc = FALSE;
  516.         if (!rc) fOk = FALSE;
  517.     }
  518.  
  519.     if (sr) CloseServiceHandle ( sr );
  520.  
  521.     return fOk;
  522. }
  523.  
  524. BOOL LoadDriverNT::Delete()
  525. {
  526.     SC_HANDLE  sr;
  527.     BOOL fOk = TRUE;
  528.  
  529.     if (m_fUseReg)
  530.         return DeleteReg();
  531.  
  532.     sr = OpenService(m_h ,m_sDriverName,SERVICE_ALL_ACCESS );
  533.  
  534.     if (sr==NULL) 
  535.         fOk = FALSE;
  536.  
  537.     if (sr)
  538.     {
  539.         DWORD rc = DeleteService( sr );
  540.  
  541.         if ( ! rc)
  542.         {
  543.             switch( GetLastError() )
  544.             {
  545.             case ERROR_ACCESS_DENIED:// The specified handle was not opened with DELETE access.
  546.                 break ;
  547.             case ERROR_SERVICE_MARKED_FOR_DELETE:// The specified service has already been marked for deletion
  548.                 rc = TRUE;
  549.                 break;
  550.             }
  551.         }
  552.  
  553.         if (!rc) fOk = FALSE;
  554.     }
  555.     
  556.     if (sr) CloseServiceHandle ( sr );
  557.  
  558.     return fOk;
  559. }
  560.   
  561. LoadDriver *NewLoadDriver(BOOL fVxd)
  562. {
  563.     OSVERSIONINFO lVerInfo;
  564.     BOOL fIsWindows98orLater;
  565.     lVerInfo.dwOSVersionInfoSize = sizeof (lVerInfo);
  566.     if (!GetVersionEx (&lVerInfo)) 
  567.     {
  568.         //printf ("Cannot get OS version\n");
  569.         return NULL;
  570.     }
  571.  
  572.     fIsWindows98orLater = (lVerInfo.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS) &&
  573.                       ((lVerInfo.dwMajorVersion > 4) || ((lVerInfo.dwMajorVersion == 4) && (lVerInfo.dwMinorVersion > 0)));
  574.  
  575.  
  576.     switch (lVerInfo.dwPlatformId)
  577.     {
  578.         case VER_PLATFORM_WIN32_NT:
  579.             if (fVxd)
  580.             {
  581.                 //printf ("-vxd is not supported on WinNT");
  582.                 break;
  583.             }
  584.             return new LoadDriverNT();
  585.             break;
  586.  
  587.         case VER_PLATFORM_WIN32_WINDOWS:
  588.             if(fIsWindows98orLater && !fVxd)
  589.                 return new LoadDriver98();
  590.             else
  591.                 return new LoadDriver95();
  592.             break;
  593.  
  594.         default:
  595.             //printf ("This program can run only on Win95/Win98/WinNT/2000");
  596.             break;
  597.     }    
  598.     return NULL;
  599. }
  600.  
  601.